package q1282_groupThePeople;

import java.util.*;

public class Solution_1 {
    /*
    暴力方法 利用一个队列hash表来遍历一次数组 遇到相同size的就都放到队列里面
    然后遍历map的所有key 如果遇到该队列长度大于key的 就分拆成多个list放入答案
    否则就直接放入答案
     */
    public List<List<Integer>> groupThePeople(int[] groupSizes) {
        Map<Integer, LinkedList<Integer>> map = new HashMap<Integer, LinkedList<Integer>>();
        for (int i = 0; i < groupSizes.length; i++) {
            if (map.containsKey(groupSizes[i])) map.get(groupSizes[i]).offer(i);
            else {
                LinkedList<Integer> list = new LinkedList<>();
                list.offer(i);
                map.put(groupSizes[i], list);
            }
        }
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        for (int size : map.keySet()) {
            if (map.get(size).size() > size) {
                int count = map.get(size).size() / size;
                for (int i = 0; i < count; i++) {
                    List<Integer> list = new ArrayList<Integer>();
                    LinkedList<Integer> temp = map.get(size);
                    for (int j = 0; j < size; j++) {
                        list.add(temp.poll());
                    }
                    res.add(new ArrayList<>(list));
                }
            } else {
                res.add(map.get(size));
            }
        }
        return res;
    }
}
